# -*- coding: utf-8 -*-
"""
@Time    : 2024/7/16 14:01 
@Author  : ZhangShenao 
@File    : 1.使用ConversationChain对话链.py 
@Desc    : 使用ConversationChain对话链

ConversationChain主要用于实现对话机制
它提供了包含Human和AI前缀的对话列表
并且内置了{history}聊天历史占位符

"""
import os

import dotenv
from langchain.chains.conversation.base import ConversationChain
from langchain_openai import ChatOpenAI

# 加载环境变量
dotenv.load_dotenv()

# 创建LLM
llm = ChatOpenAI(model_name='gpt-3.5-turbo', openai_api_base=os.getenv('OPENAI_API_BASE'))

# 创建ConversationChain对话链
conversation_chain = ConversationChain(llm=llm)

# 打印对话模版
print(conversation_chain.prompt.template)
